# coding=utf-8
import datetime # added
from django.db import models

class Question(models.Model):
    question_text = models.CharField(max_length=200)
    pub_date = models.DateTimeField('date published')
    
    def __str__(self):              # __unicode__ on Python 2
        return self.question_text
        
    # def was_published_recently(self):
    #     # return self.pub_date >= timezone.now() - datetime.timedelta(days=1) # changed
    #     return self.pub_date >= datetime.datetime.now() - datetime.timedelta(days=1)
      
    # new in tutorial05
    def was_published_recently(self):
        # now = timezone.now()
        now = datetime.datetime.now() # changed
        return now - datetime.timedelta(days=1) <= self.pub_date <= now
        
    was_published_recently.admin_order_field = 'pub_date'
    was_published_recently.boolean = True
    was_published_recently.short_description = 'Published recently?'


class Choice(models.Model):
    question = models.ForeignKey(Question)
    choice_text = models.CharField(max_length=200)
    votes = models.IntegerField(default=0)
    
    def __str__(self):              # __unicode__ on Python 2
        return self.choice_text

